'''
Created on 15/03/2012

@author: nicolas
'''

import pylab as p
from matplotlib.pyplot import plot

P = 3
Q = 2.3456

N = 30
STEP = 1
T_0 = 1.0 / 50
W_0 = 2 * p.pi / T_0
T_s = T_0 * Q / P


resultado = []

resultReal = []
resultImag = []
resultMod = []
resultFase = []

serie = p.arange(0,N,STEP)

for n in serie:
    x_d = p.exp(1j * W_0 * n * T_s)
    resultado.append(x_d)

for x in resultado:
    resultReal.append(p.real(x))
    resultImag.append(p.imag(x))
    resultMod.append(p.absolute(x))
    resultFase.append(p.angle(x))

p.figure(1)
p.subplot(211)
p.title('Real')
plot(serie, resultReal)

p.subplot(212)
p.title('Imaginario')
plot(serie, resultImag)
p.savefig('ej9_real_imag_p'+str(P)+'_q'+str(Q)+'_n'+str(N)+'.png')

p.figure(2)
p.subplot(211)
p.title('Modulo')
plot(serie, resultMod)

p.subplot(212)
p.title('Fase')
plot(serie, resultFase)
p.savefig('ej9_mod_fase_p'+str(P)+'_q'+str(Q)+'_n'+str(N)+'.png')

p.show()
